


# 测试成功！返回结果如下：
# 原文本: 这是一个测试句子。
# 嵌入向量长度: 1024
# 嵌入向量前5个值: [0.0123, -0.0456, 0.0789, -0.1234, 0.5678]


import requests

# 定义服务器地址和端口
BASE_URL = "http://0.0.0.0:50072"

def test_embedding_endpoint():
    # 测试的文本
    test_sentence = "这是一个测试句子。"
    
    # 构造请求数据
    data = {
        "text": test_sentence
    }
    
    try:
        # 发送 POST 请求到 /embedding 接口
        response = requests.post(f"{BASE_URL}/embedding", json=data)
        
        # 检查响应状态码
        if response.status_code == 200:
            # 解析返回的 JSON 数据
            result = response.json()
            print("测试成功！返回结果如下：")
            print(f"原文本: {result['text']}")
            print(f"嵌入向量长度: {len(result['embedding'])}")
            print(f"嵌入向量前5个值: {result['embedding'][:5]}")
        else:
            print(f"请求失败，状态码: {response.status_code}")
            print(f"错误信息: {response.text}")
    
    except Exception as e:
        print(f"请求过程中发生错误: {e}")

if __name__ == "__main__":
    test_embedding_endpoint()